--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
tests/backend/test_contacts_custom_image.py fda9187e95d8b9cc8373c1de23cefce6928c241d (fda9187e) Text, 2.06 KB
Tff7b72import T7ee787os
Tff7b72import T7ee787pytest
Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787databaseT7ee787.T7ee787contacts Tff7b72import Te6edf3ContactsDAO
Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787databaseT7ee787.T7ee787provider Tff7b72import Te6edf3DatabaseProvider
Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787databaseT7ee787.T7ee787schema Tff7b72import Te6edf3DatabaseSchema
Tf0883e@pytestTff7b72.Td2a8fffixture
Tff7b72def Td2a8ffdb_providerTb4b4b4(Tb4b4b4)Tb4b4b4:
Te6edf3db_path Tff7b72= Ta5d6ff"Ta5d6fftest_contacts.dbTa5d6ff"
Tff7b72if Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffexistsTb4b4b4(Te6edf3db_pathTb4b4b4)Tb4b4b4:
Te6edf3osTff7b72.Td2a8ffremoveTb4b4b4(Te6edf3db_pathTb4b4b4)
Te6edf3provider Tff7b72= Te6edf3DatabaseProviderTb4b4b4(Te6edf3db_pathTb4b4b4)
Te6edf3schema Tff7b72= Te6edf3DatabaseSchemaTb4b4b4(Te6edf3providerTb4b4b4)
Te6edf3schemaTff7b72.Td2a8ffinitializeTb4b4b4(Tb4b4b4)
Tff7b72yield Te6edf3provider
Te6edf3providerTff7b72.Td2a8ffcloseTb4b4b4(Tb4b4b4)
Tff7b72if Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffexistsTb4b4b4(Te6edf3db_pathTb4b4b4)Tb4b4b4:
Te6edf3osTff7b72.Td2a8ffremoveTb4b4b4(Te6edf3db_pathTb4b4b4)
Tff7b72def Td2a8fftest_contacts_with_custom_imageTb4b4b4(Te6edf3db_providerTb4b4b4)Tb4b4b4:
Te6edf3contacts_dao Tff7b72= Te6edf3ContactsDAOTb4b4b4(Te6edf3db_providerTb4b4b4)
T8b949e# Test adding contact with image
Te6edf3contacts_daoTff7b72.Td2a8ffadd_contactTb4b4b4(
Te6edf3nameTff7b72=Ta5d6ff"Ta5d6ffTest ContactTa5d6ff"Tb4b4b4,
Te6edf3remote_identity_hashTff7b72=Ta5d6ff"Ta5d6ffabc123def456Ta5d6ff"Tb4b4b4,
Te6edf3custom_imageTff7b72=Ta5d6ff"Ta5d6ffdata:image/png;base64,mockdataTa5d6ff"Tb4b4b4,
Tb4b4b4)
Te6edf3contact Tff7b72= Te6edf3contacts_daoTff7b72.Td2a8ffget_contact_by_identity_hashTb4b4b4(Ta5d6ff"Ta5d6ffabc123def456Ta5d6ff"Tb4b4b4)
Tff7b72assert Te6edf3contact Tff7b72is Tff7b72not Tff7b72None
Tff7b72assert Te6edf3contactTb4b4b4[Ta5d6ff"Ta5d6ffnameTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffTest ContactTa5d6ff"
Tff7b72assert Te6edf3contactTb4b4b4[Ta5d6ff"Ta5d6ffcustom_imageTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffdata:image/png;base64,mockdataTa5d6ff"
T8b949e# Test updating contact image
Te6edf3contacts_daoTff7b72.Td2a8ffupdate_contactTb4b4b4(
Te6edf3contactTb4b4b4[Ta5d6ff"Ta5d6ffidTa5d6ff"Tb4b4b4]Tb4b4b4,
Te6edf3custom_imageTff7b72=Ta5d6ff"Ta5d6ffdata:image/png;base64,updateddataTa5d6ff"Tb4b4b4,
Tb4b4b4)
Te6edf3contact Tff7b72= Te6edf3contacts_daoTff7b72.Td2a8ffget_contactTb4b4b4(Te6edf3contactTb4b4b4[Ta5d6ff"Ta5d6ffidTa5d6ff"Tb4b4b4]Tb4b4b4)
Tff7b72assert Te6edf3contactTb4b4b4[Ta5d6ff"Ta5d6ffcustom_imageTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffdata:image/png;base64,updateddataTa5d6ff"
T8b949e# Test removing contact image
Te6edf3contacts_daoTff7b72.Td2a8ffupdate_contactTb4b4b4(Te6edf3contactTb4b4b4[Ta5d6ff"Ta5d6ffidTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3clear_imageTff7b72=Tff7b72TrueTb4b4b4)
Te6edf3contact Tff7b72= Te6edf3contacts_daoTff7b72.Td2a8ffget_contactTb4b4b4(Te6edf3contactTb4b4b4[Ta5d6ff"Ta5d6ffidTa5d6ff"Tb4b4b4]Tb4b4b4)
Tff7b72assert Te6edf3contactTb4b4b4[Ta5d6ff"Ta5d6ffcustom_imageTa5d6ff"Tb4b4b4] Tff7b72is Tff7b72None
Tff7b72def Td2a8fftest_contacts_upsert_imageTb4b4b4(Te6edf3db_providerTb4b4b4)Tb4b4b4:
Te6edf3contacts_dao Tff7b72= Te6edf3ContactsDAOTb4b4b4(Te6edf3db_providerTb4b4b4)
T8b949e# Initial add
Te6edf3contacts_daoTff7b72.Td2a8ffadd_contactTb4b4b4(Ta5d6ff"Ta5d6ffUserTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffhash1Ta5d6ff"Tb4b4b4, Te6edf3custom_imageTff7b72=Ta5d6ff"Ta5d6ffimg1Ta5d6ff"Tb4b4b4)
Te6edf3contact Tff7b72= Te6edf3contacts_daoTff7b72.Td2a8ffget_contact_by_identity_hashTb4b4b4(Ta5d6ff"Ta5d6ffhash1Ta5d6ff"Tb4b4b4)
Tff7b72assert Te6edf3contactTb4b4b4[Ta5d6ff"Ta5d6ffcustom_imageTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffimg1Ta5d6ff"
T8b949e# Upsert with different image
Te6edf3contacts_daoTff7b72.Td2a8ffadd_contactTb4b4b4(Ta5d6ff"Ta5d6ffUserTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffhash1Ta5d6ff"Tb4b4b4, Te6edf3custom_imageTff7b72=Ta5d6ff"Ta5d6ffimg2Ta5d6ff"Tb4b4b4)
Te6edf3contact Tff7b72= Te6edf3contacts_daoTff7b72.Td2a8ffget_contact_by_identity_hashTb4b4b4(Ta5d6ff"Ta5d6ffhash1Ta5d6ff"Tb4b4b4)
Tff7b72assert Te6edf3contactTb4b4b4[Ta5d6ff"Ta5d6ffcustom_imageTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffimg2Ta5d6ff"
──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────